#!/bin/bash
# run_super_resolution.sh

# 设置工作目录
cd /home/vscode/workspace

# 激活Python环境（如果有虚拟环境）
# source venv/bin/activate

# 运行超分辨率训练脚本
# 使用nohup和&实现后台运行，输出重定向到日志文件
nohup python3 PyTorch_SuperResolution_GPU.py \
    --epochs 1000 \
    --batch_size 32 \
    --lr 0.001 \
    --checkpoint_interval 10 \
    --log_interval 50 \
    > training_log_$(date +%Y%m%d_%H%M%S).txt 2>&1 &

# 显示进程信息
echo "训练任务已在后台启动，PID: $!"
echo "日志文件: training_log_$(date +%Y%m%d_%H%M%S).txt"